package com.neoCalf.neoCommon.redis;

import lombok.Data;
import lombok.extern.slf4j.Slf4j;
import org.springframework.data.redis.core.RedisTemplate;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
import java.util.concurrent.TimeUnit;
import java.util.stream.Collectors;


@Slf4j
@Data
public class RedisTool<V> {
    private RedisTemplate<String, V> redisTemplate;

    /**
     * 构造函数
     *
     * @param template RedisTemplate
     */
    public RedisTool(RedisTemplate<String, V> template) {
        this.redisTemplate = template;
    }

    /**
     * 批量删除对应的value
     *
     * @param key 键
     */
    public void delete(String... key) {
        if (key != null && key.length > 0) {
            if (key.length == 1) {
                redisTemplate.delete(key[0]);
            } else {
                redisTemplate.delete(Arrays.stream(key).collect(Collectors.toList()));
            }
        }
    }

    /**
     * 获取缓存
     *
     * @param key 键
     * @return 值
     */
    public V get(String key) {
        return key == null ? null : redisTemplate.opsForValue().get(key);
    }

    /**
     * 存入缓存
     *
     * @param key   键
     * @param value 值
     */
    public void set(String key, V value) {
        try {
            redisTemplate.opsForValue().set(key, value);
        } catch (Exception ignore) {
            log.error("设置redis失败 Key:{} value:{}", key, value);
        }
    }

    /**
     * 带有时效性存入缓存
     *
     * @param key   键
     * @param value 值
     * @param time  失效
     */
    public void setX(String key, V value, long time) {
        try {
            if (time > 0) {
                redisTemplate.opsForValue().set(key, value, time, TimeUnit.SECONDS);
            } else {
                set(key, value);
            }
        } catch (Exception e) {
            log.error("设置redis失败 Key:{} value:{}", key, value);
        }
    }

    /**
     * 设置失效时间
     *
     * @param key  键
     * @param time 失效
     */
    public void expire(String key, long time) {
        try {
            if (time > 0) {
                redisTemplate.expire(key, time, TimeUnit.SECONDS);
            }
        } catch (Exception e) {
            log.error(e.getMessage());
        }
    }

    /**
     * 获取队列
     *
     * @param key 键
     * @return 队列
     */
    public List<V> getQueuelist(String key) {
        return redisTemplate.opsForList().range(key, 0, -1);
    }

    /**
     * 从队列中取出一个元素
     *
     * @param key 键
     * @return 队列中的一个元素
     */
    public V rPop(String key) {
        return redisTemplate.opsForList().rightPop(key);
    }

    /**
     * 加入队列
     *
     * @param key   键
     * @param value 值
     * @param time  失效
     */
    public void lPush(String key, V value, long time) {
        try {
            redisTemplate.opsForList().leftPush(key, value);
            if (time > 0) {
                expire(key, time);
            }
        } catch (Exception e) {
            log.error(e.getMessage());
        }
    }

    /**
     * 从队列中取出一个元素
     *
     * @param key 键
     * @return 队列中的一个元素
     */
    public Long remove(String key, Object value) {
        try {
            return redisTemplate.opsForList().remove(key, 1, value);
        } catch (Exception e) {
            log.error(e.getMessage());
            return 0L;
        }
    }

    /**
     * 向集合中添加元素
     *
     * @param key   键
     * @param value 值
     */
    public void addSet(String key, V value) {
        redisTemplate.opsForSet().add(key, value);
    }

    /**
     * 移除集合中的元素
     *
     * @param key   键
     * @param value 值
     */
    public void removeSet(String key, V value) {
        redisTemplate.opsForSet().remove(key, value);
    }

    /**
     * 获取集合
     *
     * @param key 键
     * @return 集合
     */
    public List<V> getSet(String key) {
        Long size = redisTemplate.opsForSet().size(key);
        if (size != null && size > 0) {
            return redisTemplate.opsForSet().randomMembers(key, size);
        } else {
            return new ArrayList<>();
        }
    }

    /**
     * 获取集合中的一个元素
     *
     * @param key 键
     * @return 集合中的一个元素
     */
    public V getSetOne(String key) {
        if (getSet(key) != null && !getSet(key).isEmpty()) {
            return getSet(key).get(0);
        }
        return null;
    }

    /**
     * 批量存入缓存
     *
     * @param key    键
     * @param values 值
     * @param time   失效
     */
    public void lPushAll(String key, List<V> values, long time) {
        try {
            redisTemplate.opsForList().leftPushAll(key, values);
            if (time > 0) {
                expire(key, time);
            }
        } catch (Exception e) {
            log.error("批量存入redis失败:\n", e);
        }
    }

    /**
     * 哈希表加入
     *
     * @param key 键
     * @param hk  哈希表字段
     * @param hv  哈希表值
     */
    public void putHash(String key, String hk, Object hv) {
        try {
            redisTemplate.opsForHash().put(key, hk, hv);
        } catch (Exception e) {
            log.error("缓存哈希加入失败:\n", e);
        }
    }

    /**
     * 获取哈希表中指定字段的值
     *
     * @param key 键
     * @param hk  哈希表字段
     * @return 哈希表中指定字段的值
     */
    public Object getHash(String key, String hk) {
        try {
            return redisTemplate.opsForHash().get(key, hk);
        } catch (Exception e) {
            log.error("缓存哈希获取失败:\n", e);
        }
        return null;
    }

    /**
     * 获取键的剩余过期时间
     *
     * @param key 键
     * @return 剩余过期时间（秒）
     */
    public int ttl(String key) {
        Long t = redisTemplate.getExpire(key);
        t = t == null ? 0 : t;
        return Math.toIntExact(t);
    }
}
